# 注:processed_data.csv数据文件本地没有,但平台已经内置;squarify包本地未安装,但平台已经内置
# ⚠️ 平台原始代码 - 请原样输入至教学平台(注释除外),平台才会判定答案正确
import pandas as pd # 导入Pandas数据分析库
import numpy as np # 导入NumPy数值计算库
import matplotlib.pyplot as plt # 导入Matplotlib绘图库
import seaborn as sns # 导入Seaborn可视化库
import warnings # 导入warnings模块用于控制警告输出
warnings.filterwarnings('ignore') # 忽略警告
plt.rcParams['font.sans-serif'] = ['SimHei'] # 中文显示
plt.rcParams['axes.unicode_minus'] = False # 负号显示
data = pd.read_csv('processed_data.csv') # 读取数据
# 产品分类映射
category_mapping = {'安溪铁观音':'茶叶','武夷岩茶':'茶叶','福州茉莉花':'茶叶','古田银耳':'食用菌','建宁莲子':'中药材','琯溪蜜柚':'水果','宁德大黄鱼':'水产品'}
data['category'] = data['product_name'].map(category_mapping).fillna(data['category']) # 更新产品分类
print(data.head()) # 查看数据
# 产品分析
import squarify # 导入树形图库
# 创建季节特征(复用之前的季节映射)
seasons = {1: '冬季', 2: '冬季', 3: '春季', 4: '春季', 5: '春季', 6: '夏季', 7: '夏季', 8: '夏季', 9: '秋季', 10: '秋季', 11: '秋季', 12: '冬季'}
data['下单月份'] = pd.to_datetime(data['order_date']).dt.month # 提取月份
data['order_season'] = data['下单月份'].map(seasons) # 映射季节
def product_analysis(orders): # 产品分析函数
# 按商品大类统计销售数据
category_stats = data.groupby('category').agg({'order_id':'count','sales_amount':'sum'}).reset_index() #【要求1】
category_stats.columns = ['产品类型', '订单数', '销售额'] # 重命名列
category_stats['平均订单金额'] = category_stats['销售额'] / category_stats['订单数'] # 计算平均订单金额
category_stats = category_stats.sort_values('销售额', ascending=False) # 按销售额排序
# 显示商品大类统计
print("产品类型统计:")
print(category_stats) # 输出统计量数据
# 可视化商品大类销售额
plt.figure(figsize=(14, 8))
sns.barplot(x='产品类型', y='销售额', data=category_stats, palette='viridis') # 绘制分类柱状图
plt.title('各商品大类销售额') # 设置图表标题
plt.xlabel('产品类型') # 设置X轴标签
plt.ylabel('销售额(元)') # 设置Y轴标签
plt.grid(axis='y') # 显示网格线
plt.tight_layout() # 自动调整布局防止重叠
plt.savefig("商品大类销售额.png") # 保存图表
# 按商品子类统计销售数据
subcategory_stats = data.groupby(['category', 'product_name']).agg({'order_id':'count','sales_amount':'sum'}).reset_index()
subcategory_stats.columns = ['产品类型', '产品子类', '订单数', '销售额'] # 重命名列
subcategory_stats['平均订单金额'] = subcategory_stats['销售额'] / subcategory_stats['订单数'] # 计算平均订单金额
subcategory_stats = subcategory_stats.sort_values('销售额', ascending=False) # 按销售额排序
# 显示Top3商品子类
print("产品子类统计:")
print(subcategory_stats.head(3)) # 输出前几行数据
# 可视化Top3商品子类销售额
top_subcategories = subcategory_stats.head(3)
plt.figure(figsize=(16, 10)) # 创建图形画布
sns.barplot(x='销售额', y='产品子类', hue='产品类型', data=top_subcategories, palette='viridis') # 绘制分类柱状图
plt.title('各产品子类销售额Top 3') # 设置图表标题
plt.xlabel('销售额(元)') # 设置X轴标签
plt.ylabel('产品子类') # 设置Y轴标签
plt.grid(axis='x') # 显示网格线
plt.tight_layout() # 自动调整布局防止重叠
plt.savefig("Top3子类销售额.png") # 保存图表
# 商品定价分析
data['价格段'] = pd.cut(data['price'], bins=[0, 100, 200, 500], labels=['0-100元', '100-200元', '200-500元']) #【要求2】
# 按价格段统计销售数据
price_range_stats = data.groupby('价格段').agg({'order_id':'count','sales_amount':'sum'}).reset_index()
price_range_stats.columns = ['价格段', '订单数', '销售额'] # 重命名列
price_range_stats['订单占比'] = price_range_stats['订单数'] / price_range_stats['订单数'].sum() * 100 # 计算订单占比
price_range_stats['销售额占比'] = price_range_stats['销售额'] / price_range_stats['销售额'].sum() * 100 # 计算销售额占比
# 显示价格段统计
print("价格段统计:")
print(price_range_stats) # 输出价格数据
# 可视化价格段销售情况
fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(16, 14))
# 订单数占比柱状图
sns.barplot(x='价格段', y='订单占比', data=price_range_stats, palette='Blues_d', ax=ax1)
ax1.set_title('各价格段订单数占比') # 设置图表标题
ax1.set_xlabel('价格段') # 设置X轴标签
ax1.set_ylabel('订单数占比(%)') # 设置Y轴标签
ax1.grid(axis='y') # 设置网格线
# 销售额占比柱状图
sns.barplot(x='价格段', y='销售额占比', data=price_range_stats, palette='Reds_d', ax=ax2)
ax2.set_title('各价格段销售额占比') # 设置图表标题
ax2.set_xlabel('价格段') # 设置X轴标签
ax2.set_ylabel('销售额占比(%)') # 设置Y轴标签
ax2.grid(axis='y') # 设置网格线
plt.tight_layout() # 自动调整布局防止重叠
plt.savefig("价格段销售占比.png") # 保存图表
# 商品销售分布树形图
plt.figure(figsize=(16, 12))
category_sales = category_stats[['产品类型', '销售额']].copy() # 创建数据副本
category_sales['标签'] = category_sales['产品类型'] + '\n' + category_sales['销售额'].apply(lambda x: f'{x:,.0f}元') # 生成标签
squarify.plot(sizes=category_sales['销售额'], label=category_sales['标签'], alpha=0.8, color=sns.color_palette("viridis", len(category_sales))) # 绘制树形图
plt.axis('off') # 隐藏坐标轴
plt.title('各产品类型销售额树形图', fontsize=18) # 设置图表标题
plt.tight_layout() # 自动调整布局防止重叠
plt.savefig("产品销售额树形图.png") # 保存图表
# 商品季节性分析
seasonal_product = data.groupby(['order_season', 'category'])['sales_amount'].sum().reset_index() # 按季节和产品类型统计销售额
season_pivot = seasonal_product.pivot(index='category', columns='order_season', values='sales_amount') # 【要求3】
season_pct = season_pivot.div(season_pivot.sum(axis=1), axis=0) * 100 # 计算各季节占比
# 显示季节性销售占比
print("季节商品销售占比(%):")
print(season_pct) # 输出百分比数据
# 可视化季节性销售占比
plt.figure(figsize=(14, 10))
sns.heatmap(season_pct, annot=True, fmt='.1f', cmap='YlGnBu') # 绘制热力图
plt.title('各产品类型在不同季节的销售占比(%)') # 设置图表标题
plt.xlabel('季节') # 设置X轴标签
plt.ylabel('产品类型') # 设置Y轴标签
plt.tight_layout() # 自动调整布局防止重叠
plt.savefig("产品季节销售占比.png") # 保存图表
print("产品分析完成") # 输出产品分析完成
# 进行产品分析
product_analysis(data)